library(lintr)
library(xml2)
library(microbenchmark)

# Generate a file with 10,000 function calls to emphasize full-tree scan costs
generate_large_file <- function(n_calls = 10000) {
  lines <- vapply(1:n_calls, function(i) sprintf("f%d(a = %d, b = %d)", i, i, i), character(1L))
  sprintf("{\n%s\n}", paste(lines, collapse = "\n"))
}

text <- generate_large_file(10000)
temp <- tempfile()
writeLines(text, temp)
source_exprs <- get_source_expressions(temp)
xml <- source_exprs$full_xml_parsed_content %||% source_exprs$expressions[[1L]]$xml_parsed_content

bench_orig <- function() {
  res <- xml_find_all(xml, "//SYMBOL_FUNCTION_CALL/parent::expr")
  length(res)
}

bench_optimized <- function() {
  res <- xml_parent(xml_find_all(xml, "//SYMBOL_FUNCTION_CALL"))
  length(res)
}

cat("Benchmark (1) R/source_utils.R (Full-tree Cache Initialization):\n")
print(microbenchmark(orig = bench_orig(), optimized = bench_optimized(), times = 50))
unlink(temp)

library(lintr)
library(xml2)
library(glue)
library(microbenchmark)

# Setup
except <- c("mutate", "transmute")
ns_calls <- lintr:::xp_text_in_table(except)
xpath_orig <- glue("//EQ_SUB[not(preceding-sibling::expr/SYMBOL_FUNCTION_CALL[{ ns_calls }])]/parent::expr[count(EQ_SUB) > 1]")
xpath_hybrid_base <- glue("//EQ_SUB[not(preceding-sibling::expr/SYMBOL_FUNCTION_CALL[{ ns_calls }])]")

generate_large_file <- function(n_calls = 2000) {
  lines <- vapply(1:n_calls, function(i) {
    if (i %% 10 == 0) sprintf("list(a%d = 1, a%d = 2)", i, i)
    else sprintf("list(a%d = 1, b%d = 2)", i, i)
  }, character(1L))
  sprintf("{\n%s\n}", paste(lines, collapse = "\n"))
}

text <- generate_large_file(2000)
temp <- tempfile()
writeLines(text, temp)
source_exprs <- get_source_expressions(temp)
xml <- source_exprs$full_xml_parsed_content %||% source_exprs$expressions[[1L]]$xml_parsed_content

bench_orig <- function() {
  call_expr <- xml_find_all(xml, xpath_orig)
  length(call_expr)
}

bench_hybrid <- function() {
  # Jump to parents then filter in R
  call_expr <- xml_parent(xml_find_all(xml, xpath_hybrid_base))
  call_expr <- call_expr[vapply(call_expr, function(x) length(xml_find_all(x, "EQ_SUB")) > 1L, logical(1L))]
  length(call_expr)
}

cat("\nBenchmark (2) R/duplicate_argument_linter.R (Hybrid vs. Pure XPath):\n")
print(microbenchmark(orig = bench_orig(), hybrid = bench_hybrid(), times = 50))
unlink(temp)

library(lintr)
library(xml2)
library(microbenchmark)

generate_large_file <- function(n = 5000) {
  lines <- vapply(1:n, function(i) sprintf("x <- -%d", i), character(1L))
  sprintf("{\n%s\n}", paste(lines, collapse = "\n"))
}

text <- generate_large_file(5000)
temp <- tempfile()
writeLines(text, temp)
source_exprs <- get_source_expressions(temp)
xml <- source_exprs$full_xml_parsed_content %||% source_exprs$expressions[[1L]]$xml_parsed_content
nums <- xml_find_all(xml, "//NUM_CONST")

bench_orig <- function() {
  # Pure XPath relative jump
  is_negative <- !is.na(xml_find_first(nums, "parent::expr/preceding-sibling::OP-MINUS"))
}

bench_optimized <- function() {
  # xml_parent helper jump
  is_negative <- !is.na(xml_find_first(xml_parent(nums), "preceding-sibling::OP-MINUS"))
}

cat("\nBenchmark (3) R/implicit_integer_linter.R (Nodeset Jump: helper vs. XPath):\n")
print(microbenchmark(orig = bench_orig(), optimized = bench_optimized(), times = 100))
unlink(temp)